package com.example.examsystem.repository;

import com.example.examsystem.model.UserRole;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

import java.util.List;

/**
 * 用户角色关联仓库接口
 * 用于管理用户和角色之间的多对多关系
 */
@Repository
public interface UserRoleRepository extends JpaRepository<UserRole, Long> {
    
    /**
     * 根据用户ID查找用户角色关系
     */
    List<UserRole> findByUserId(Long userId);
    
    /**
     * 根据角色ID查找用户角色关系
     */
    List<UserRole> findByRoleId(Long roleId);
    
    /**
     * 根据用户ID和角色ID查找用户角色关系
     */
    UserRole findByUserIdAndRoleId(Long userId, Long roleId);
    
    /**
     * 根据用户ID删除用户角色关系
     */
    @Modifying
    @Transactional
    void deleteByUserId(Long userId);
    
    /**
     * 根据角色ID删除用户角色关系
     */
    @Modifying
    @Transactional
    void deleteByRoleId(Long roleId);
    
    /**
     * 根据用户ID和角色ID删除用户角色关系
     */
    @Modifying
    @Transactional
    void deleteByUserIdAndRoleId(Long userId, Long roleId);
    
    /**
     * 检查用户是否拥有指定角色
     */
    boolean existsByUserIdAndRoleId(Long userId, Long roleId);
    
    /**
     * 根据用户ID查询角色ID列表
     */
    @Query("SELECT ur.roleId FROM UserRole ur WHERE ur.userId = :userId")
    List<Long> findRoleIdsByUserId(@Param("userId") Long userId);
    
    /**
     * 根据角色ID查询用户ID列表
     */
    @Query("SELECT ur.userId FROM UserRole ur WHERE ur.roleId = :roleId")
    List<Long> findUserIdsByRoleId(@Param("roleId") Long roleId);
}


